@echo off
echo ========================================
echo      AI服务环境变量设置
echo ========================================
echo.
echo 当前AI模型状态：
echo.

REM 检查通义千问API密钥
if defined QWEN_API_KEY (
    echo ✅ 通义千问：已配置 ^(QWEN_API_KEY^)
) else if defined DASHSCOPE_API_KEY (
    echo ✅ 通义千问：已配置 ^(DASHSCOPE_API_KEY^)
) else (
    echo ❌ 通义千问：未配置
)

REM 检查DeepSeek API密钥
if defined DEEPSEEK_API_KEY (
    echo ✅ DeepSeek：已配置
) else (
    echo ❌ DeepSeek：未配置
)

REM 检查OpenAI API密钥
if defined OPENAI_API_KEY (
    echo ✅ OpenAI：已配置
) else (
    echo ❌ OpenAI：未配置
)

echo.
echo ========================================
echo      设置API密钥
echo ========================================
echo.
echo 请选择要配置的AI模型：
echo 1. 通义千问 ^(推荐，免费额度^)
echo 2. DeepSeek ^(性价比高^)
echo 3. OpenAI ^(功能最强^)
echo 4. 跳过设置
echo.

set /p choice=请输入选择 ^(1-4^)：

if "%choice%"=="1" goto setup_qwen
if "%choice%"=="2" goto setup_deepseek
if "%choice%"=="3" goto setup_openai
if "%choice%"=="4" goto skip_setup
echo 无效选择，请重新运行脚本
pause
exit /b 1

:setup_qwen
echo.
echo 配置通义千问API密钥：
echo 1. 访问：https://dashscope.console.aliyun.com/
echo 2. 登录阿里云账号
echo 3. 创建API密钥
echo.
set /p qwen_key=请输入通义千问API密钥：

if "%qwen_key%"=="" (
    echo 未输入API密钥，跳过设置
    goto skip_setup
)

setx QWEN_API_KEY "%qwen_key%"
echo ✅ 通义千问API密钥已设置
echo ⚠️ 请重新打开命令行窗口使环境变量生效
goto end

:setup_deepseek
echo.
echo 配置DeepSeek API密钥：
echo 1. 访问：https://platform.deepseek.com/
echo 2. 注册/登录账号
echo 3. 创建API密钥
echo.
set /p deepseek_key=请输入DeepSeek API密钥：

if "%deepseek_key%"=="" (
    echo 未输入API密钥，跳过设置
    goto skip_setup
)

setx DEEPSEEK_API_KEY "%deepseek_key%"
echo ✅ DeepSeek API密钥已设置
echo ⚠️ 请重新打开命令行窗口使环境变量生效
goto end

:setup_openai
echo.
echo 配置OpenAI API密钥：
echo 1. 访问：https://platform.openai.com/api-keys
echo 2. 登录账号
echo 3. 创建API密钥
echo.
set /p openai_key=请输入OpenAI API密钥：

if "%openai_key%"=="" (
    echo 未输入API密钥，跳过设置
    goto skip_setup
)

setx OPENAI_API_KEY "%openai_key%"
echo ✅ OpenAI API密钥已设置
echo ⚠️ 请重新打开命令行窗口使环境变量生效
goto end

:skip_setup
echo 跳过AI密钥设置，将使用基础规则回答
echo 如需高质量AI回答，请稍后运行此脚本配置API密钥

:end
echo.
echo ========================================
echo      启动AI服务
echo ========================================
echo.
echo 请选择操作：
echo 1. 立即启动AI服务 ^(当前窗口^)
echo 2. 新窗口启动AI服务
echo 3. 仅设置环境变量，稍后手动启动
echo.

set /p start_choice=请输入选择 ^(1-3^)：

if "%start_choice%"=="1" (
    echo 正在启动AI服务...
    python main.py
) else if "%start_choice%"=="2" (
    echo 在新窗口启动AI服务...
    start cmd /k "python main.py"
    echo AI服务已在新窗口启动
) else (
    echo 环境变量设置完成
    echo 稍后请运行：python main.py
)

echo.
pause
















